# scripts/loss_plot.py
import pandas as pd
import matplotlib.pyplot as plt

# 读取CSV文件
df = pd.read_csv('../loss.csv', usecols=[1, 2], skiprows=100)

# 确保列名为 'step' 和 'value'
# 如果列名不同，可以使用 df.columns = ['step', 'value'] 来重命名
df.columns = ['Step', 'Value']

# 绘制损失曲线图
plt.figure(figsize=(10, 6))
plt.plot(df['Step'], df['Value'], label='Loss', color='blue', linewidth=2)

# 添加标题和标签
plt.title('Training Loss Curve', fontsize=16)
plt.xlabel('Step', fontsize=12)
plt.ylabel('Loss Value', fontsize=12)

# 添加网格
plt.grid(True, linestyle='--', alpha=0.5)

# 显示图例
plt.legend()

# 优化布局
plt.tight_layout()

# 显示图像
plt.show()